package com.web.common.tool;

import org.springframework.http.MediaType;
import org.springframework.util.StringUtils;

public class HttpMediaType {
	
	private HttpMediaType() {}

	/**
	 * 根据文件后缀获取header类型
	 * @param fileType 文件类型
	 * @return Content-Type对应类型
	 *
	 * @author zhouhui
	 * @since 1.0.0
	 */
	public static MediaType getMediaType(String fileType) {
		MediaType mediaType = MediaType.APPLICATION_OCTET_STREAM;
		if(!StringUtils.hasText(fileType)) {
			return mediaType;
		}
		fileType = fileType.toLowerCase();
		if("png".equals(fileType)) {
			mediaType = MediaType.IMAGE_PNG;
		}else if("gif".equals(fileType)) {
			mediaType = MediaType.IMAGE_GIF;
		}else if("jpeg".equals(fileType) || "jpg".equals(fileType)) {
			mediaType = MediaType.IMAGE_JPEG;
		}else if("pdf".equals(fileType)) {
			mediaType = MediaType.APPLICATION_PDF;
		}else if("ico".equals(fileType)) {
			mediaType = new MediaType("image", "x-icon");
		}
		return mediaType;
	}
}
